#pragma once
#include <cuda_runtime.h>
#define _TRT_INFER_BEGIN namespace trt_infer {
#define _TRT_INFER_END }

#ifdef NDEBUG
#define _ITERATOR_DEBUG_LEVEL 2
#else
#define _ITERATOR_DEBUG_LEVEL 1
#endif

#if (defined _WIN32 || defined WIN32) && defined PI_API_EXPORTS
#  define PI_EXPORTS __declspec(dllexport)
#elif defined __GNUC__ && __GNUC__ >= 4
#  define PI_EXPORTS __attribute__ ((visibility ("default")))
#else
#  define PI_EXPORTS __declspec(dllimport)
#endif

struct CUstream_st;
typedef CUstream_st CUStreamRaw;
typedef cudaStream_t CUStream;
//typedef CUStreamRaw *CUStream;

#define CURRENT_DEVICE_ID           -1
